package com.lvan.commonutils.exechandler;

import com.lvan.commonutils.response.Result;
import com.lvan.commonutils.response.ResultCode;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.ResponseBody;

/**
 * 全局异常处理
 *
 * @author LuoMeng
 */
@ControllerAdvice
@Slf4j
public class GlobalExceptionHandler {

    /**
     * 全局异常处理
     *
     * @param e exception
     * @return Result
     */
    @ExceptionHandler(Exception.class)
    @ResponseBody
    public Result error(Exception e) {
        log.error(e.getMessage());
        return Result.error().message("异常:系统繁忙请稍后重试，请求系统管理员帮助吧!");
    }

    /**
     * 特定异常处理 - 算术异常
     *
     * @param e exception
     * @return Result
     */
    @ExceptionHandler(ArithmeticException.class)
    @ResponseBody
    public Result error(ArithmeticException e) {
        log.error(e.getMessage());
        return Result.error().code(ResultCode.ARITHMETIC_EXCEPTION.getCode())
                .message(ResultCode.ARITHMETIC_EXCEPTION.getMessage());
    }

    /**
     * 自定义异常处理
     *
     * @param e 自定义异常类
     * @return Result
     */
    @ExceptionHandler(CustomException.class)
    @ResponseBody
    public Result error(CustomException e) {
        log.error(e.getMessage());
        return Result.error().code(e.getCode()).message(e.getMessage());
    }
}
